from app import create_app, db
from app.models import User
from datetime import datetime

def create_admin():
    app = create_app()
    with app.app_context():
        # 检查管理员是否已存在
        admin = User.query.filter_by(username='admin').first()
        if admin:
            print('管理员账号已存在！')
            return
        
        # 创建管理员账号
        admin = User(
            username='admin',
            email='admin@w850.com',
            is_active=True,
            is_admin=True,
            created_at=datetime.utcnow()
        )
        admin.set_password('admin123')  # 设置初始密码
        
        # 保存到数据库
        db.session.add(admin)
        db.session.commit()
        print('管理员账号创建成功！')
        print('用户名: admin')
        print('密码: admin123')
        print('请登录后立即修改密码！')

if __name__ == '__main__':
    create_admin() 